import matplotlib.pyplot as plt
import numpy as np

plt.rcParams['font.sans-serif']=[u'simHei']

# 数据
X = np.array([10,15,20,30,50,60,60,70])
Y = np.array([0.8,1.0,1.8,2.0,3.2,3.0,3.1,3.5])


# 线性回归
coefficients = np.polyfit(X,Y,1)


a,b = coefficients

Y_fit=a * X + b

# 绘图
# 创建一个新图形
plt.figure()
#原始数据点
plt.plot(X,Y,'o')
# 线性回归线
plt.plot(X,Y_fit,'-')
# 标签
plt.xlabel('房屋面积（m^2）')
plt.ylabel('租赁价格（1000元）')
plt.title('房屋面积和租赁价格线性回归')
plt.legend(['数据点','线性回归线'])

# 显示图形
plt.show()
